tensor([[0.5900, 0.5900, 0.5900, 0.5900, 0.5900],
        [0.4100, 0.4100, 0.4100, 0.4100, 0.4100]], grad_fn=<RoundBackward1>)